package util

import (
	"crypto/rand"
	"io"
)

// 生成验证码 加入生成6位

// 通过下面的数组生成数字验证码
var randowCodes = [...]byte{
	'1', '2', '3', '4', '5', '6', '7', '8', '9', '0',
}

func GenValidateCode(max int) string {
	b := make([]byte, max)

	io.ReadAtLeast(rand.Reader, b, max)
	for i := 0; i < max; i++ {
		b[i] = randowCodes[int(b[i])%len(randowCodes)]
	}
	return string(b)
}
